from langchain.chains import ConversationChain
from langchain.memory import ConversationSummaryBufferMemory
from tool import langchain_client
from langchain.chains import ConversationChain

if __name__ == '__main__':
    # 创建一个长字符串
    schedule = "在八点你和你的产品团队有一个会议。 \ 你需要做一个PPT。 \ 上午9点到12点你需要忙于LangChain。\ Langchain是一个有用的工具，因此你的项目进展的非常快。\ 中午，在意大利餐厅与一位开车来的顾客共进午餐 \ 走了一个多小时的路程与你见面，只为了解最新的 AI。 \ 确保你带了笔记本电脑可以展示最新的 LLM 样例."
    memory = ConversationSummaryBufferMemory(llm=langchain_client, max_token_limit=100)
    memory.save_context({"input": "你好，我叫皮皮鲁"}, {"output": "你好啊，我叫鲁西西"})
    memory.save_context({"input": "很高兴和你成为朋友!"}, {"output": "是的，让我们一起去冒 险吧!"})
    memory.save_context({"input": "今天的日程安排是什么?"}, {"output": f"{schedule}"})
    print(memory.load_memory_variables({})['history'])

    conversation = ConversationChain(llm=langchain_client, memory=memory, verbose=True)
    conversation.predict(input="展示什么样的样例最好呢?")

    print(memory.load_memory_variables({}))  # 摘要记录更新了